from rest_framework.response import Response


# 返回特定的格式，前台得到数据
# 继承自Response
class APIResponse(Response):
    # 当你发送一个APIResponse对象时
    # 比如从一个视图函数或类视图中返回它
    # Django REST framework 会自动将这个对象转换为 HTTP 响应，并且通常情况下会是一个 JSON 响应。
    def __init__(self, code=0, msg='', data=None, status=200, headers=None, content_type=None, **kwargs):
        dic = {'code': code, 'msg': msg}
        if data is not None:
            dic['data'] = data

        dic.update(kwargs)  # 这里使用update
        super().__init__(data=dic, status=status,
                         template_name=None, headers=headers,
                         exception=False, content_type=content_type)
